from megaparse.Converter import MegaParse
from quivr_api.models.files import File

from .common import process_file


def process_pdf(
    file: File,
    brain_id,
    original_file_name,
    integration=None,
    integration_link=None,
):
    return process_file(
        file=file,
        loader_class=MegaParse,
        brain_id=brain_id,
        original_file_name=original_file_name,
        integration=integration,
        integration_link=integration_link,
    )
